package com.atg.ai_agent.rag;


/*
author: atg
time: 2025/10/3 19:24
*/

import org.springframework.ai.chat.prompt.PromptTemplate;
import org.springframework.ai.rag.generation.augmentation.ContextualQueryAugmenter;


/**
 * 创建上下文查询增强器  -- 用于处理知识库中没有查询到的情况
 */

public class LoveContextualQueryAugmenter {
    public static ContextualQueryAugmenter create() {

        PromptTemplate promptTemplate = new PromptTemplate("""
                你是一个AI助手，如果为查询到相关的文档，请使用以下模板：
                抱歉，我无法找到与您查询相关的信息。有问题请联系atg
                
                """);

      return ContextualQueryAugmenter.builder()
                .emptyContextPromptTemplate(promptTemplate)  // 如果在知识库中没有查询到的话 ，使用这个模板
                .allowEmptyContext(false)
                .build();

    }
}
